как написать генератор на python